package com.davidlife.security.webattack;

import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.Arrays;

/**
 * 通过魔数来判断文件类型
 */
public class ReadMagicNumberAndGetFileType {

    public static void main(String[] args) throws IOException {
        String filePath = "D:\\workspace\\IdeaProjects\\knowledge-framework-demo\\spring-basic\\notes-images\\spring-overview.png";
        System.out.println(getFileTypeByMagicNumberCode(filePath));
        System.out.println(getFileTypeByFileTypeName(filePath));

        System.out.println(Arrays.toString(hexString2Bytes(getFileHeader(filePath))));
    }

    /**
     * 根据读取的魔数，判断文件类型
     */
    public static FileTypeEnum getFileTypeByMagicNumberCode(String filePath) throws IOException {
        String fileHead = getFileHeader(filePath);
        System.out.println(fileHead);
        if (fileHead == null || fileHead.length() == 0) {
//            return FileTypeEnum.NOT_EXITS_ENUM;
            return null;
        }
        fileHead = fileHead.toUpperCase();
        FileTypeEnum[] fileTypes = FileTypeEnum.values();
        for (FileTypeEnum fileType : fileTypes) {
            if (fileHead.startsWith(fileType.getMagicNumberCode())) {
                return fileType;
            }
        }
        return null;
    }

    /**
     * 根据文件的扩展名，判断文件类型
     */
    public static FileTypeEnum getFileTypeByFileTypeName(String filePath) throws IOException {
        String fileTypeName = filePath.substring(filePath.lastIndexOf(".") + 1);
        if (fileTypeName == null || fileTypeName.length() == 0) {
//            return FileTypeEnum.NOT_EXITS_ENUM;
            return null;
        }
        FileTypeEnum[] fileTypes = FileTypeEnum.values();
        for (FileTypeEnum fileType : fileTypes) {
            if (fileTypeName.equalsIgnoreCase(fileType.getFileTypeName())) {
                return fileType;
            }
        }
        return null;
    }

    /**
     * 读取文件头
     * 注意：每个文件的魔数的长度都不相同，所以这里只是读取文件头，而没有说读取多长。
     * 这里暂且读取头部28个字节，应该够用了。且要后面要用 startWith 判断
     */
    public static String getFileHeader(String filePath) throws IOException {
        //这里需要注意的是，每个文件的魔数的长度都不相同，因此需要使用 startWith
        byte[] bytes = new byte[28];
        InputStream is = new FileInputStream(filePath);
        is.read(bytes, 0, 28);
        is.close();
        System.out.println(Arrays.toString(bytes));
        return bytes2HexString(bytes);
    }

    /**
     * 将二进制信息（一个byte数组）转换为 16进制 字符串 表示
     * 也可说是 将字节编码为16进制表示
     * Convert byte[] to hex string.这里我们可以将byte转换成int，然后利用Integer.toHexString(int)来转换成16进制字符串。
     */
    public static String bytes2HexString(byte[] bytes) {
        StringBuilder hexStr = new StringBuilder();
        if (bytes == null || bytes.length == 0) {
            return null;
        }
        for (int i = 0; i < bytes.length; i++) {
            int intv = bytes[i] & 0xFF;
            String strv = Integer.toHexString(intv);
            if (strv.length() < 2) {
                hexStr.append(0);
            }
            hexStr.append(strv);
        }
        return hexStr.toString();
    }

    /**
     * 将 16进制的字符串 转换为 二进制存储（一个byte数组）
     * 也可说是 将16进制表示解码为字节
     * Convert hex string to byte[]
     */
    public static byte[] hexString2Bytes(String hexStr) {
        if (hexStr == null || hexStr.equals("")) {
            return null;
        }
        hexStr = hexStr.toUpperCase();
        int length = hexStr.length() / 2;
        char[] hexChars = hexStr.toCharArray();
        byte[] bytes = new byte[length];
        for (int i = 0; i < length; i++) {
            int pos = i * 2;
            bytes[i] = (byte) (charToByte(hexChars[pos]) << 4 | charToByte(hexChars[pos + 1]));
        }
        return bytes;
    }

    private static byte charToByte(char c) {
        return (byte) "0123456789ABCDEF".indexOf(c);
    }

}
